Description

When PGMQueue is initialized with async DSN/engine/session_maker, the _check_pgmq_ext() and _check_pg_partman_ext() methods were attempting to use sync context managers with AsyncSession objects, causing TypeError: 'AsyncSession' object does not support the context manager protocol.

Changes

Following feedback to maintain strict separation between sync and async operations, the fix was implemented on the test side:

• Fixed _check_pgmq_ext() and _check_pg_partman_ext(): Added async detection logic that routes to async versions (_check_pgmq_ext_async(), _check_pg_partman_ext_async()) when is_async is True
• Created async versions of check methods: Prevents nested event loop issues by using await in async contexts
• Updated async methods: create_partitioned_queue_async() and drop_queue_async() now use await when calling async check methods
• Created test helper utility: Added call_method() in tests/_async_utils.py that automatically routes to sync or async methods based on the PGMQueue instance type
• Updated test fixtures and tests: All fixtures (pgmq_setup_teardown, pgmq_partitioned_setup_teardown) and test methods now properly handle both sync and async PGMQueue instances using the call_method() helper

This approach maintains the architectural separation between _execute_operation (sync) and _execute_async_operation (async) while ensuring all tests work correctly with both sync and async PGMQueue instances.

Example

# Tests now work with both sync and async PGMQueue instances
def test_create_queue(pgmq_all_variants, db_session):
    pgmq = pgmq_all_variants
    queue_name = f"test_queue_{uuid.uuid4().hex}"
    # Automatically calls create_queue() or create_queue_async() based on pgmq.is_async
    call_method(pgmq, 'create_queue', queue_name)
    assert check_queue_exists(db_session, queue_name) is True

Status

• In progress
• Ready for review
• Done

Checklist

• Read the Contributing Guide
• Passes tests
• Linted ( we use pre-commit with ruff )
• Updated documentation

💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.